package demo.socket.demo1;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;

/**
 * 聊天室服务端
 */
public class Server {

    private ServerSocket server;

    private Collection<PrintWriter> allOut = new ArrayList<>();


    public Server(){
        try{

            System.out.println("正在启动服务端-------");
            server = new ServerSocket(8088);
            System.out.println("服务端启动完毕!!");
        }catch (IOException e){
            e.printStackTrace();
        }
    }

    public void start(){
        try {
            while (true) {
                System.out.println("正在连接客户端-----");
                Socket socket = server.accept();
                System.out.println("一个客户端已连接!");

                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server server = new Server();
        server.start();
    }
    private class ClientHandler implements Runnable{
        private Socket socket;
        private String host;
        public ClientHandler(Socket socket){
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();
        }
        public void run(){
            PrintWriter pw = null;
            try {
                BufferedReader br = new BufferedReader(
                        new InputStreamReader(
                                socket.getInputStream()
                        )
                );
                pw = new PrintWriter(
                        new BufferedWriter(
                                new OutputStreamWriter(
                                        socket.getOutputStream(),"UTF-8"
                                )
                        ),true
                );
            synchronized (allOut){
                allOut.add(pw);
            }
            sendMessage(host+"上线了!当前在线人数:"+allOut.size());
            String line;
            while ((line = br.readLine())!=null){
                System.out.println(host+":"+line);
                sendMessage(host+":"+line);
            }
            } catch (IOException e) {

            }finally {
                synchronized (allOut){
                    allOut.remove(pw);
                }
                sendMessage(host+":下线了!");
                try {
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        public void sendMessage(String message){
            synchronized (allOut){
                for (PrintWriter out :allOut){
                    out .println(message);
                }
            }
        }


    }
















}
